#pragma once

#include <iostream>
#include <unistd.h>
#include <sys/wait.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include "../comm/util.hpp"
#include "../comm/log.hpp"

// 只负责进行代码的编译

namespace ns_compiler
{
    // 引入路径拼接功能
    using namespace ns_util;
    using namespace ns_log;

    class Compiler
    {
    public:
        Compiler()
        {}
        ~Compiler()
        {}
        //返回值：编译成功：true，否则：false
        //输入参数：编译的文件名
        //file_name: 1234
        //1234 -> ./temp/1234.cpp
        //1234 -> ./temp/1234.exe
        //1234 -> ./temp/1234.stderr
                static bool Compile(const std::string &filename) // 这里的file_name就是单纯的文件名，不带有路径和后缀
        {
            int pid = fork();
            if (pid < 0)
            {
                LOG(ERROR)<<"创建子进程失败！"<<std::endl;
                return false;
            }
            else if (pid == 0)
            {
                // 打开stderr文件，要将子进程编译之后，编译错误的文件重定向输出到CompilerError文件中,
                int _stderr = open(PathUtil::CompilerError(filename).c_str(), O_CREAT | O_WRONLY, 0644);
                if (_stderr < 0)
                {
                    LOG(ERROR)<<"创建stderr文件失败!"<<std::endl;
                    exit(1);
                }
                // 重定向,将输出到标准错误的内容，全部重定向输出到stderr文件中
                dup2(_stderr, 2);

                // 这是子进程，子进程进行编译
                // 子进程进行进程替换函数，执行其它代码
                // 子进程要对源文件进行编译，就必须得有源文件，所以得先通过文件名获取带文件路径和后缀的源文件
                execlp("g++", "g++", "-o", PathUtil::Exe(filename).c_str(), PathUtil::Src(filename).c_str(),"-D", "COMPILER_ONLINE", "-std=c++11", nullptr);
                LOG(ERROR)<<"编译失败,请检查g++命令行参数传递是否符合规则!"<<std::endl;
                exit(2);
            }
            else
            {
                // 这是父进程，父进程要等待子进程
                waitpid(pid, nullptr, 0);
                // 判断子进程是否编译成功，即判断对应的路径下是否生成可执行程序
                if (FileUtil::IsFileExists(PathUtil::Exe(filename)))
                {
                    LOG(INFO)<< PathUtil::Src(filename)<<"编译成功！"<<std::endl;
                    return true;
                }
            }
            LOG(ERROR)<<"没有形成可执行程序，编译失败！"<<std::endl;
            return false;
        }
    };
}